import unittest
from core.game import Game

class GameTest(unittest.TestCase):
    
    def setUp(self):
        self.game=Game(5);
    
    def testGameCreate(self):
        self.assertEqual(5, self.game.amount)
        self.assertEqual('STARTED', self.game.status)
        
    def testSheriffIsDead(self):
        self.game.onSheriffGoToDeatch()
        self.assertEqual('ENDED', self.game.status)
    
    
if __name__=='__main__':
    unittest.main()